package com.atguigu.tingshu.user.service.impl;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.lang.UUID;
import com.atguigu.tingshu.common.constant.RedisConstant;
import com.atguigu.tingshu.common.constant.SystemConstant;
import com.atguigu.tingshu.common.rabbit.constant.MqConst;
import com.atguigu.tingshu.common.rabbit.service.RabbitService;
import com.atguigu.tingshu.model.user.UserListenProcess;
import com.atguigu.tingshu.user.service.UserListenProcessService;
import com.atguigu.tingshu.vo.album.TrackStatMqVo;
import com.atguigu.tingshu.vo.album.TrackStatVo;
import com.atguigu.tingshu.vo.user.UserListenProcessVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@SuppressWarnings({"all"})
public class UserListenProcessServiceImpl implements UserListenProcessService {

	@Autowired
	private MongoTemplate mongoTemplate;

	@Autowired
	private RedisTemplate redisTemplate;

	@Autowired
	private RabbitService rabbitService;


	/**
	 * 获取声音的上次跳出时间
	 * 从MongoDB查询，根据用户ID+声音ID查询上次播放秒数 从MongoDB查询，根据用户ID+声音ID查询上次播放秒数
	 * @param trackId
	 * @return
	 */
	@Override
	public BigDecimal getTrackBreakSecond(Long trackId,Long userId) {

		//程序每隔一段时间就会将endTime存入mongoDB中,我们要做的就是得到那个时间

		//从mongoDB中得到endTime

		//1.根据用户ID确定操作集合名称 形式为：固定前缀_用户ID=userListenProcess_userId
		String collectionName = this.getCollectionName(userId);

		//2.查询条件：用户ID+声音ID
		Query query = new Query();
		query.addCriteria(Criteria.where("userId").is(userId).and("trackId").is(trackId));

		//3.执行查询MongoDB
		UserListenProcess listenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);

		if (listenProcess != null){
			BigDecimal breakSecond = listenProcess.getBreakSecond();
			return breakSecond;
		}

		return BigDecimal.valueOf(0);
	}


	/**
	 * 更新声音播放进度
	 *
	 * @param userListenProcessVo
	 * @return
	 */
	@Override
	public void updateListenProcess(UserListenProcessVo userListenProcessVo,Long userId) {

		//首先获取CollectionName
		String collectionName = this.getCollectionName(userId);

		//2.根据用户ID+声音ID查询用户播放进度
		Query query = new Query();
		query.addCriteria(Criteria.where("userId").is(userId).and("trackId").is(userListenProcessVo.getTrackId()));

		UserListenProcess listenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);


		//设置播放秒数小数位，舍入模式
		userListenProcessVo.setBreakSecond(userListenProcessVo.getBreakSecond().setScale(0, RoundingMode.DOWN));

		if (listenProcess == null){
			//3.播放进度不存在，构建播放进度，保存到MongoDB
			listenProcess = new UserListenProcess();

			listenProcess.setUserId(userId);
			listenProcess.setAlbumId(userListenProcessVo.getAlbumId());
			listenProcess.setTrackId(userListenProcessVo.getTrackId());
			listenProcess.setBreakSecond(userListenProcessVo.getBreakSecond());
			listenProcess.setCreateTime(new Date());
			listenProcess.setUpdateTime(new Date());


		}else {
			//4.播放进度存在，更新播放秒+更新时间
			listenProcess.setBreakSecond(userListenProcessVo.getBreakSecond());
			listenProcess.setUpdateTime(new Date());
		}

		mongoTemplate.save(listenProcess,collectionName);


		//5.TODO 基于MQ可靠性消息，更新声音/专辑(播放量)在MySQL以及ES中统计数值

		//5.1 当天内判断当前用户是否为第一次收听更新统计数值 核心：生产者生产消息幂等性
		//5.1.1 关键点找出业务标识作为RedisKey：前缀:用户ID_专辑ID_声音ID

		String key = RedisConstant.USER_TRACK_REPEAT_STAT_PREFIX + "_"+userId+"_"+userListenProcessVo.getAlbumId()+"_"+userListenProcessVo.getTrackId();

		//5.1.2 采用Redis提供set k v ex nx  命令，设置RedisKey，设置过期时间，写入成功则发送MQ消息，更新统计数值；写入失败则忽略

		//设置过期时间:要求,一天内不论听多少次声音,只加一次播放量
		Date date = new Date();
		long ttl = DateUtil.endOfDay(date).getTime() - date.getTime();
		Boolean flag = redisTemplate.opsForValue().setIfAbsent(key, "1", ttl, TimeUnit.MILLISECONDS);

		//5.2 发送MQ消息，广播消息（通知"专辑服务"更新DB，通知"搜索服务"更新ES）
		if (flag){
			//5.2.1走到这说明添加分布式锁成功,执行业务逻辑
			TrackStatMqVo mqVo = new TrackStatMqVo();
			mqVo.setBusinessNo(UUID.randomUUID().toString());
			mqVo.setAlbumId(userListenProcessVo.getAlbumId());
			mqVo.setTrackId(userListenProcessVo.getTrackId());
			mqVo.setStatType(SystemConstant.TRACK_STAT_PLAY);
			mqVo.setCount(1);

			//5.2.2 发送MQ消息
			rabbitService.sendMessage(MqConst.EXCHANGE_TRACK,MqConst.ROUTING_TRACK_STAT_UPDATE,mqVo);

		}
	}


	/**
	 * 获取用户最近一次播放记录
	 * @return
	 */
	@Override
	public Map<String, Long> getLatelyTrack(Long userId) {

		HashMap<String, Long> map = new HashMap<>();

		//首先获取collectionName
		String collectionName = this.getCollectionName(userId);

		//从mongoDB中获取根据更新时间降序获取到最新听的声音

		Query query = new Query();

		query.with(Sort.by(Sort.Direction.DESC,"updateTime"));
		query.limit(1);

		UserListenProcess listenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);

		if (listenProcess != null){
			//走到这说明有数据
			map.put("trackId",listenProcess.getTrackId());
			map.put("albumId",listenProcess.getAlbumId());
			return map;
		}

		return null;
	}

	private String getCollectionName(Long userId) {

		String collectionName = "userListenProcess_" + userId;

		return collectionName;
	}
}
